#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/shape_inference.h"
#include "tensorflow/core/framework/op_kernel.h"
using namespace tensorflow;


REGISTER_OP("CustomAssignaddVariable")
    .Input("var: resource")
    .Input("value: float")
    .SetIsStateful();

class CustomAssignaddVariableOp : public OpKernel {
 public:
  explicit CustomAssignaddVariableOp(OpKernelConstruction* context) : OpKernel(context) {}

  void Compute(OpKernelContext* context) override {
    OP_REQUIRES_OK(context, errors::Unimplemented("CustomAssignaddVariableOp is not supported on CPU"));
  }
};

REGISTER_KERNEL_BUILDER(Name("CustomAssignaddVariable").Device(DEVICE_CPU), CustomAssignaddVariableOp);